package io.renren.modules.doc.service;

import com.baomidou.mybatisplus.extension.service.IService;
import io.renren.common.utils.PageUtils;
import io.renren.modules.doc.entity.DenseRagEntity;

import java.util.List;
import java.util.Map;

/**
 * 向量检索服务
 */
public interface DenseRagService extends IService<DenseRagEntity> {

    /**
     * 分页查询
     */
    PageUtils queryPage(Map<String, Object> params);
    
    /**
     * 获取详情
     */
    DenseRagEntity getDetail(Long id);
    
    /**
     * 删除
     */
    void deleteBatch(Long[] ids);
    
    /**
     * 向量相似度检索
     */
    List<DenseRagEntity> searchSimilar(String query, int limit);
    
    /**
     * 生成文本的向量嵌入
     * @param text 输入文本
     * @return 向量嵌入数据
     */
    byte[] generateEmbeddings(String text);
}
